How to simplify this simple JS script